#pragma once
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include "Log.hpp"
#include "Thread.hpp"
#include "Cond.hpp"
#include "Mutex.hpp"

namespace ThreadPoolModule
{
    using namespace ThreadModlue;
    using namespace LogModule;
    using namespace CondModule;
    using namespace MutexModule;

    static const int gnum = 5;

    template <typename T>
    class ThreadPool
    {
    private:
        // 唤醒所有休眠线程
        void WakeUpAllThread()
        {
            LockGuard lockguard(_mutex); // 加锁保护共享数据
            if (_sleepernum)             // 如果有休眠线程
                _cond.Broadcast();       // 广播唤醒所有等待线程
            LOG(LogLevel::INFO) << "唤醒所有的休眠线程";
        }
        // 唤醒单个休眠线程
        void WakeUpOne()
        {
            _cond.Signal(); // 发送信号唤醒一个等待线程
            LOG(LogLevel::INFO) << "唤醒一个休眠线程";
        }

        // 私有构造函数（单例模式）
        ThreadPool(int num = gnum)
            : _num(num),         // 初始化线程数量
              _isrunning(false), // 初始状态为未运行
              _sleepernum(0)     // 初始休眠线程数为0
        {
            // 创建指定数量的工作线程
            for (int i = 0; i < num; i++)
            {
                _threads.emplace_back([this]()
                                      {
                                          HandlerTask(); // 每个线程执行任务处理函数
                                      });
            }
        }
        // 启动线程池
        void Start()
        {
            if (_isrunning)
                return;        // 避免重复启动
            _isrunning = true; // 设置运行标志

            // 启动所有线程
            for (auto &thread : _threads)
            {
                thread.Start(); // 启动线程
                LOG(LogLevel::INFO) << "start new thread success: " << thread.Name();
            }
        }
        // 禁止拷贝构造和赋值
        ThreadPool(const ThreadPool<T> &) = delete;
        ThreadPool<T> &operator=(const ThreadPool<T> &) = delete;

    public:
        // 获取单例实例(线程安全)
        static ThreadPool<T> *GetInstance()
        {
            if (inc == nullptr)
            {                               // 第一次检查（避免不必要的加锁）
                LockGuard lockguard(_lock); // 加锁保护

                LOG(LogLevel::DEBUG) << "获取单例....";
                if (inc == nullptr)
                { // 第二次检查（确保单例唯一）
                    LOG(LogLevel::DEBUG) << "首次使用单例, 创建之....";
                    inc = new ThreadPool<T>(); // 创建实例
                    inc->Start();              // 启动线程池
                }
            }
            return inc; // 返回单例指针
        }
        
        // 停止线程池
        void Stop()
        {
            if (!_isrunning)
                return;         // 避免重复停止
            _isrunning = false; // 设置停止标志

            WakeUpAllThread(); // 唤醒所有线程以便退出
        }
        // 等待所有线程结束
        void Join()
        {
            for (auto &thread : _threads)
            {
                thread.Join(); // 等待每个线程结束
            }
        }
        void HandlerTask()
        {
            char name[128];
            pthread_getname_np(pthread_self(), name, sizeof(name));

            while (true)
            {
                T t;

                { // 临界区
                    LockGuard lockGuard(_mutex);

                    while (_taskq.empty() && _isrunning)
                    {
                        _sleepernum++;      // 增加休眠线程计数
                        _cond.Wait(_mutex); // 等待条件变量
                        _sleepernum--;      // 被唤醒后减少计数
                    }

                    // 检查退出条件：线程池停止且任务队列为空
                    if (!_isrunning && _taskq.empty())
                    {
                        LOG(LogLevel::INFO) << name << " 退出了, 线程池退出&&任务队列为空";
                        break; // 退出线程循环
                    }

                    // 获取任务(此时队列非空)
                    t = _taskq.front(); // 取出任务
                        _taskq.pop();
                }

                t();
            }
        }
        bool Enqueue(const T &in)
        {
            if (_isrunning)
            {
                LockGuard lockguard(_mutex);
                _taskq.push(in);

                // 如果所以线程都在休眠
                if (_threads.size() == _sleepernum)
                    WakeUpOne();

                return true;
            }
            return false;
        }
        // 析构函数
        ~ThreadPool()
        {
            // 注意：单例对象通常需要手动释放
        }

    private:
        std::vector<Thread> _threads;
        int _num;
        std::queue<T> _taskq;
        Cond _cond;
        Mutex _mutex;

        bool _isrunning; // 线程池运行状态
        int _sleepernum;

        static ThreadPool<T> *inc; // 单例实例指针
        static Mutex _lock;        // 单例创建锁（解决双重检查锁定问题）
    };

    // 静态成员初始化
    template <typename T>
    ThreadPool<T> *ThreadPool<T>::inc = nullptr;

    template <typename T>
    Mutex ThreadPool<T>::_lock;

}